package org.kuaishou.util;

import org.junit.Test;

import java.net.MalformedURLException;
import java.net.URL;

public class URLUtil {
    /**
     * 根据输入的URL字符串规范化URL对象。
     * 如果输入的URL字符串不包含协议（例如 "http://"），则添加 "http://" 后返回URL对象。
     *
     * @param urlString 输入的URL字符串
     * @return 规范化后的URL对象
     * @throws MalformedURLException 如果URL字符串无效，则抛出MalformedURLException
     */
    public static URL normalizeURL(String urlString) throws MalformedURLException {
        // 如果URL字符串不包含协议，则添加 "http://"
        if (urlString.startsWith("//")) {
            return new URL("http:" + urlString);
        }
        if (!urlString.startsWith("http://") && !urlString.startsWith("https://")) {
            return new URL("http://" + urlString);
        }
        // 创建并返回URL对象
        return new URL(urlString);
    }


    @Test
    public void t1() throws Exception {
        URL url = URLUtil.normalizeURL("//sf6-cdn-tos.douyinstatic.com/obj/tos-cn-ve-2774/okIBCInhecoGOE5h6ZvqCBYtfXCIMQEbgkRKgD");
        URL url2 = URLUtil.normalizeURL("https://p6.douyinpic.com/aweme/200x200/tos-cn-v-2774c002/3e4a28d0da714f67b4d4ccaa9eba6827.jpeg");
        URL url3 = URLUtil.normalizeURL("121/212");
        System.out.println(url);
        System.out.println(url2);
        System.out.println(url3);
    }
}
